#include <stdio.h>
#include <stdlib.h>
#include <iostream>
#include <sys/stat.h>
#include <time.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <string.h>
//#include <sqlite3.h>
#include <sys/socket.h>
#include <net/if.h> // struct ifreq
#include <sys/ioctl.h>
#include <uuid/uuid.h>
#include <vector>
#include "myfileinfo.h"
#include <algorithm>
#include <stdio.h>
#include <string.h>
#include "baseInfo.h"	// 基础信息
#include<stdlib.h>
using namespace std;
#define F(x, y, z) (((x) & (y)) | ((~x) & (z)))

#define G(x, y, z) (((x) & (z)) | ((y) & (~z)))

#define H(x, y, z) ((x) ^ (y) ^ (z))

#define I(x, y, z) ((y) ^ ((x) | (~z)))

 

#define RL(x, y) (((x) << (y)) | ((x) >> (32 - (y))))

 

#define PP(x) (x<<24)|((x<<8)&0xff0000)|((x>>8)&0xff00)|(x>>24)

 

#define FF(a, b, c, d, x, s, ac) a = b + (RL((a + F(b,c,d) + x + ac),s))

#define GG(a, b, c, d, x, s, ac) a = b + (RL((a + G(b,c,d) + x + ac),s))

#define HH(a, b, c, d, x, s, ac) a = b + (RL((a + H(b,c,d) + x + ac),s))

#define II(a, b, c, d, x, s, ac) a = b + (RL((a + I(b,c,d) + x + ac),s))

 

unsigned A,B,C,D,a,b,c,d,i,len,flen[2],x[16];

char filename[200];

FILE *fp;

#define MD5_LEN 32
void md5()

{

 

    a=A,b=B,c=C,d=D;

    /**//* Round 1 */

    FF (a, b, c, d, x[ 0], 7, 0xd76aa478); /**//* 1 */

    FF (d, a, b, c, x[ 1], 12, 0xe8c7b756); /**//* 2 */

    FF (c, d, a, b, x[ 2], 17, 0x242070db); /**//* 3 */

    FF (b, c, d, a, x[ 3], 22, 0xc1bdceee); /**//* 4 */

    FF (a, b, c, d, x[ 4], 7, 0xf57c0faf); /**//* 5 */

    FF (d, a, b, c, x[ 5], 12, 0x4787c62a); /**//* 6 */

    FF (c, d, a, b, x[ 6], 17, 0xa8304613); /**//* 7 */

    FF (b, c, d, a, x[ 7], 22, 0xfd469501); /**//* 8 */

    FF (a, b, c, d, x[ 8], 7, 0x698098d8); /**//* 9 */

    FF (d, a, b, c, x[ 9], 12, 0x8b44f7af); /**//* 10 */

    FF (c, d, a, b, x[10], 17, 0xffff5bb1); /**//* 11 */

    FF (b, c, d, a, x[11], 22, 0x895cd7be); /**//* 12 */

    FF (a, b, c, d, x[12], 7, 0x6b901122); /**//* 13 */

    FF (d, a, b, c, x[13], 12, 0xfd987193); /**//* 14 */

    FF (c, d, a, b, x[14], 17, 0xa679438e); /**//* 15 */

    FF (b, c, d, a, x[15], 22, 0x49b40821); /**//* 16 */

 

    /**//* Round 2 */

    GG (a, b, c, d, x[ 1], 5, 0xf61e2562); /**//* 17 */

    GG (d, a, b, c, x[ 6], 9, 0xc040b340); /**//* 18 */

    GG (c, d, a, b, x[11], 14, 0x265e5a51); /**//* 19 */

    GG (b, c, d, a, x[ 0], 20, 0xe9b6c7aa); /**//* 20 */

    GG (a, b, c, d, x[ 5], 5, 0xd62f105d); /**//* 21 */

    GG (d, a, b, c, x[10], 9, 0x02441453); /**//* 22 */

    GG (c, d, a, b, x[15], 14, 0xd8a1e681); /**//* 23 */

    GG (b, c, d, a, x[ 4], 20, 0xe7d3fbc8); /**//* 24 */

    GG (a, b, c, d, x[ 9], 5, 0x21e1cde6); /**//* 25 */

    GG (d, a, b, c, x[14], 9, 0xc33707d6); /**//* 26 */

    GG (c, d, a, b, x[ 3], 14, 0xf4d50d87); /**//* 27 */

    GG (b, c, d, a, x[ 8], 20, 0x455a14ed); /**//* 28 */

    GG (a, b, c, d, x[13], 5, 0xa9e3e905); /**//* 29 */

    GG (d, a, b, c, x[ 2], 9, 0xfcefa3f8); /**//* 30 */

    GG (c, d, a, b, x[ 7], 14, 0x676f02d9); /**//* 31 */

    GG (b, c, d, a, x[12], 20, 0x8d2a4c8a); /**//* 32 */

 

    /**//* Round 3 */

    HH (a, b, c, d, x[ 5], 4, 0xfffa3942); /**//* 33 */

    HH (d, a, b, c, x[ 8], 11, 0x8771f681); /**//* 34 */

    HH (c, d, a, b, x[11], 16, 0x6d9d6122); /**//* 35 */

    HH (b, c, d, a, x[14], 23, 0xfde5380c); /**//* 36 */

    HH (a, b, c, d, x[ 1], 4, 0xa4beea44); /**//* 37 */

    HH (d, a, b, c, x[ 4], 11, 0x4bdecfa9); /**//* 38 */

    HH (c, d, a, b, x[ 7], 16, 0xf6bb4b60); /**//* 39 */

    HH (b, c, d, a, x[10], 23, 0xbebfbc70); /**//* 40 */

    HH (a, b, c, d, x[13], 4, 0x289b7ec6); /**//* 41 */

    HH (d, a, b, c, x[ 0], 11, 0xeaa127fa); /**//* 42 */

    HH (c, d, a, b, x[ 3], 16, 0xd4ef3085); /**//* 43 */

    HH (b, c, d, a, x[ 6], 23, 0x04881d05); /**//* 44 */

    HH (a, b, c, d, x[ 9], 4, 0xd9d4d039); /**//* 45 */

    HH (d, a, b, c, x[12], 11, 0xe6db99e5); /**//* 46 */

    HH (c, d, a, b, x[15], 16, 0x1fa27cf8); /**//* 47 */

    HH (b, c, d, a, x[ 2], 23, 0xc4ac5665); /**//* 48 */

 

    /**//* Round 4 */

    II (a, b, c, d, x[ 0], 6, 0xf4292244); /**//* 49 */

    II (d, a, b, c, x[ 7], 10, 0x432aff97); /**//* 50 */

    II (c, d, a, b, x[14], 15, 0xab9423a7); /**//* 51 */

    II (b, c, d, a, x[ 5], 21, 0xfc93a039); /**//* 52 */

    II (a, b, c, d, x[12], 6, 0x655b59c3); /**//* 53 */

    II (d, a, b, c, x[ 3], 10, 0x8f0ccc92); /**//* 54 */

    II (c, d, a, b, x[10], 15, 0xffeff47d); /**//* 55 */

    II (b, c, d, a, x[ 1], 21, 0x85845dd1); /**//* 56 */

    II (a, b, c, d, x[ 8], 6, 0x6fa87e4f); /**//* 57 */

    II (d, a, b, c, x[15], 10, 0xfe2ce6e0); /**//* 58 */

    II (c, d, a, b, x[ 6], 15, 0xa3014314); /**//* 59 */

    II (b, c, d, a, x[13], 21, 0x4e0811a1); /**//* 60 */

    II (a, b, c, d, x[ 4], 6, 0xf7537e82); /**//* 61 */

    II (d, a, b, c, x[11], 10, 0xbd3af235); /**//* 62 */

    II (c, d, a, b, x[ 2], 15, 0x2ad7d2bb); /**//* 63 */

    II (b, c, d, a, x[ 9], 21, 0xeb86d391); /**//* 64 */

 

    A += a;

    B += b;

    C += c;

    D += d;

 

}

 

int CalcFileMD5(char *filename, char *md5_sum)

{

        if (filename[0]==34) filename[strlen(filename)-1]=0,strcpy(filename,filename+1);

        if (!strcmp(filename,"exit")) exit(0);

        if (!(fp=fopen(filename,"rb"))) //

        {

            printf("Can not open this file!\n");

            return 0;

        }

        fseek(fp, 0, SEEK_END);

        if((len=ftell(fp))==-1)

        {

            printf("Sorry! Can not calculate files which larger than 2 GB!\n");

            fclose(fp);

            return 0;

        }

        rewind(fp);

        A=0x67452301,B=0xefcdab89,C=0x98badcfe,D=0x10325476;

        flen[1]=len/0x20000000;

        flen[0]=(len%0x20000000)*8;

        memset(x,0,64);

        fread(&x,4,16,fp);

        for(i=0; i<len/64; i++)

        {

            md5();

            memset(x,0,64);

            fread(&x,4,16,fp);

        }

        ((char*)x)[len%64]=128;

        if(len%64>55) md5(),memset(x,0,64);

        memcpy(x+14,flen,8);

        md5();

        fclose(fp);

        sprintf(md5_sum,"%08x%08x%08x%08x",PP(A),PP(B),PP(C),PP(D));
//cout<<md5_sum<<endl;

 return 1;

}
// 替换字符串中的指定字符
// str : 需要替换的字符串
// cOld: 被替换的字符（可以是字符串）
// cNew: 新字符（可以是字符串）
// 注：原字符串地址空间必须够存替换后的字符串，否则会越界
int StrReplace_(char *str, char *cOld, char *cNew)
{
    if(str == NULL || cOld == NULL)
        return -1;
    
    int nSize1 = strlen(str);
    int nSize2 = strlen(cOld);
    int nSize3 = strlen(cNew);

    //string StrTemp;
    char StrTemp[1024] = {0};
    memset(StrTemp, 0, sizeof(char)*1024);
    int x = 0;
    int y = 0;
    for(int i = 0; i < nSize1;)
    {
        if(strncmp(str+i, cOld, nSize2) == 0)
        {
            //cout << "有" << endl;
            for(int j = 0; j < nSize3; j++)
            {
                StrTemp[y+j] = cNew[j];
            }
            i += nSize2;
            x += nSize2;
            y += nSize3;
        }
        else
        {
            StrTemp[y] = str[x];
            i++;
            x++;
            y++;
        }
    }

    StrTemp[y] = '\0';
    strcpy(str, StrTemp);
   
    return 0;
}

void TimeFormat(time_t sTime, char * fmtTime)
{
    time_t tempTime = sTime;
    struct tm* tt;
    tt = gmtime(&tempTime);
    char catime[32];
    sprintf(catime, "%04d-%02d-%02d %02d:%02d:%02d", tt->tm_year+1900, tt->tm_mon+1, tt->tm_mday, (tt->tm_hour+8)%24, tt->tm_min, tt->tm_sec);
    strcpy(fmtTime, catime);
    //cout << catime << endl;
}

void GetMac(char *cMac)
{
    int fd;
    int interfaceNum = 0;
    struct ifreq buf[16];
    struct ifconf ifc;
    struct ifreq ifrcopy;
    char mac[18] = {0};
    //char ip[32] = {0};
    //char broadAddr[32] = {0};
    //char subnetMask[32] = {0};


    if ((fd = socket(AF_INET, SOCK_DGRAM, 0)) < 0)
    {
        perror("socket");
        close(fd);
        return ;
    }

    ifc.ifc_len = sizeof(buf);
    ifc.ifc_buf = (caddr_t)buf;
    if (!ioctl(fd, SIOCGIFCONF, (char *)&ifc))
    {
        interfaceNum = ifc.ifc_len / sizeof(struct ifreq);
        //printf("interface num = %d\n", interfaceNum);
        while (interfaceNum-- > 0)
        {
            //printf("ndevice name: %s\n", buf[interfaceNum].ifr_name);
            ifrcopy = buf[interfaceNum];
            if (ioctl(fd, SIOCGIFFLAGS, &ifrcopy))
            {
                //printf("ioctl: %s [%s:%d]n", strerror(errno), __FILE__, __LINE__);
                close(fd);
                return ;
            }
            //get the mac of this interface  
            if (!ioctl(fd, SIOCGIFHWADDR, (char *)(&buf[interfaceNum])))
            {
                memset(mac, 0, sizeof(mac));
                snprintf(mac, sizeof(mac), "%02x-%02x-%02x-%02x-%02x-%02x",
                    (unsigned char)buf[interfaceNum].ifr_hwaddr.sa_data[0],
                    (unsigned char)buf[interfaceNum].ifr_hwaddr.sa_data[1],
                    (unsigned char)buf[interfaceNum].ifr_hwaddr.sa_data[2],
                    (unsigned char)buf[interfaceNum].ifr_hwaddr.sa_data[3],
                    (unsigned char)buf[interfaceNum].ifr_hwaddr.sa_data[4],
                    (unsigned char)buf[interfaceNum].ifr_hwaddr.sa_data[5]
                );
                char * failMac = "00:00:00:00:00:00";
                //cout << "mac =" << mac << endl;
                if(strcmp(mac, failMac) != 0)
                {
                    //printf("device mac: %s\n", mac);
                    memcpy(cMac, mac, strlen(mac));
                    break;
                }
            }
            else
            {
                //printf("ioctl: %s [%s:%d]n", strerror(errno), __FILE__, __LINE__);
                close(fd);
                return ;
            }

        }
    }

    close(fd);
}
char *substrend(char *str, int n)
{
	char * substr = (char*) malloc (n+1);
	int length = strlen(str);
	if(n >= length)//若截取长度大于字符串长度，则直接截取全部字符串
	{
		strcpy(substr, str);
		return substr;
	}
	int k = 0;
	for(int i = strlen(str) - n ; i < strlen(str); i++)
	{
		substr[k++] = str[i];
	}
	substr[k] = '\0';
	return substr;
}
char *fristsubstrend(char *str, int n)
{
	char * substr = (char*) malloc (n+1);
	memset(substr,0,n+1);
	int length = strlen(str);
	if(n >= length)//若截取长度大于字符串长度，则直接截取全部字符串
	{
		strcpy(substr, str);
		return substr;
	}
	int k = 0;
//	for(int i = strlen(str) - n ; i < strlen(str); i++)//去后面字符
	for(int i = 0 ; i < n; i++)//去前面字符
	{
		substr[k++] = str[i];
	}
	substr[k] = '\0';
	return substr;
}
bool GetFileInfo(char *fileName_, sqlite3 *db, vector<pair<unsigned int, char*> >&vt, char *evenId,char *Mac)
{
    	 char md5_sum[MD5_LEN + 1];
    	char *filename = fileName_;
	char* type;
	char* type1;
	char file_type[4];
	type=new char[4];
	type1=new char[4];
	char *fristpath;
	char *fristpath1;
	fristpath=new char[10];
	fristpath1=new char[5];
    /*	// Mac地址
    	char Mac[100] = {0};
 	GetMac(Mac);
	//char mac[100] = {0};
	//memset(mac, 0, sizeof(mac));
	//GetMacAddr(mac);
	char cIp[36];           // 本地ip
    	memset(cIp, 0, sizeof(cIp));
    	LocalIp(cIp, 36);
 	char mac[100];
	memset(mac,0,sizeof(mac));
	strcat(mac,Mac);
	strcat(mac,"~~");
	strcat(mac,cIp);
   */
    //char *dbname = dbName_; 
    //2020.11.8 add md5
	memset(fristpath,0,10);
	memset(fristpath1,0,5);
	fristpath=fristsubstrend(filename,8);
	fristpath1=fristsubstrend(filename,4);
//cout<<fristpath<<endl;
    if(strcasecmp(fristpath1,"/lib")==0||strcasecmp(fristpath,"/usr/lib")==0)
    {
	//cout<<fristpath<<endl;
	//cout<<fristpath1<<endl;
	memset(type,0,4);
	type=substrend(filename,3);
	
	memset(type1,0,4);

	memset(file_type,0,4);
	type1=substrend(filename,2);
        if(strcasecmp (type,".so")==0||strcasecmp (type1,".a")==0)
        {
	   if(strcasecmp(type,".so")==0)
	   {
	    strcpy(file_type,type);
	   }
	  else
	   {
	    strcpy(file_type,type1);
	   }
	//cout<<file_type<<endl;
   // UUID
        char str[36];
        uuid_t uuid;
        uuid_generate(uuid);
        uuid_unparse(uuid, str);
        string temp(str);
        temp.erase(remove(temp.begin(), temp.end(), '-'));
        memset(str, 0, sizeof(str));
        strncpy(str, temp.c_str(), 32);
        char file_name[256];
        memset(file_name,0,256);
        // 文件名
        char * sub2 = strrchr(filename, '/');
         sub2 += 1;
         //cout << sub2 << endl;
         StrReplace_(sub2, "'", "''");
        cout << sub2 << endl;
         char *err_msg;
        CalcFileMD5(filename,md5_sum);
        char sql[1024] = {0};
        sprintf(sql, "insert into linux_file_legitimacy (id, filename, filepath,file_type,file_md5,mac_address, event_task_id) values('%s', '%s', '%s', '%s','%s', '%s', '%s')",
        str, sub2, filename,file_type,md5_sum ,Mac, evenId);
    

        int ret = sqlite3_exec(db, sql, NULL, NULL, &err_msg);
        if(ret != 0)
        {
            perror(err_msg);
            sqlite3_free(err_msg);
            sqlite3_close(db);

            return false;
        }
            cout<<"md5zhi"<<md5_sum<<endl;
        }
    }
    if(type)
	    {
		delete[] type;
	    }
  	   if(type1)
	   {
		delete[] type1;
	   }
    if(fristpath)
	    {
		delete[] fristpath;
	    }
  	   if(fristpath1)
	   {
		delete[] fristpath1;
	   }
   
    struct stat fileinfo;
    stat(filename, &fileinfo);  

    time_t atime = fileinfo.st_atime;
    time_t mtime = fileinfo.st_mtime;
    time_t cctime = fileinfo.st_ctime;

    char fmt_atime[32] = {0};   // 文件最后访问时间
    char fmt_mtime[32] = {0};   // 文件内容最后修改时间
    char fmt_ctime[32] = {0};   // 文件状态改变时间
    TimeFormat(atime, fmt_atime);
    TimeFormat(mtime, fmt_mtime);
    TimeFormat(cctime, fmt_ctime);

    // 文件所有者
    char name[1024] = {0};
    bool bExist = false;
    vector<pair<unsigned int, char*> >::iterator it;
    for(it = vt.begin(); it != vt.end(); it++)
    {
        if(it->first == fileinfo.st_uid)
        {
            strcpy(name, it->second);
            bExist = true;
            break;
        }
    }

    StrReplace_(name, "'", "''");

    if(!bExist)
    {
        char  cmd[4096] = {0};
        sprintf(cmd, "cat /etc/passwd | cut -d : -f 1,3 | grep -I :%u > uid.temp", fileinfo.st_uid);

        pid_t Pid;
        Pid = fork();
        if(Pid > 0) // 父进程
        {
            int status;
            waitpid(Pid, &status, 0);
        }
        else if (Pid == 0)  // 子进程
        {
            char *Exe = "/bin/sh";
            execl(Exe, "sh", "-c", cmd, (char*)0);
        }
        else
        {
            perror("creat process failed\n");
            return false;
        }

        FILE * fp = NULL;
        char * fileName = "./uid.temp";
        fp = fopen(fileName, "r");

        bool User = true;
        char line[1024] = {0};
        if(fp != NULL)
        {
            char * str = fgets(line, 1024, fp);
            //fclose(fp);
            if(str == NULL)
            {
                User = false;
            }
        }
        else
        {
            perror("open file failed\n");
            return false;
        }
        fclose(fp);

        if(User)
        {
            char * sub = strchr(line, ':');
            strncpy(name, line, strlen(line) - strlen(sub));

            // 将
            char *cName = new char[1024];
            memset(cName, 0, sizeof(char)*1024);
            strcpy(cName, name);
            pair<unsigned int, char*> pr(fileinfo.st_uid, cName);
            vt.push_back(pr);
        }
        else
        {
            sprintf(name, "%d", fileinfo.st_uid);
        }
    }

    
    //cout << Mac << endl;

    // UUID
    char str[36];
    uuid_t uuid;
    uuid_generate(uuid);
    uuid_unparse(uuid, str);
    string temp(str);
    temp.erase(remove(temp.begin(), temp.end(), '-'));
    memset(str, 0, sizeof(str));
    strncpy(str, temp.c_str(), 32);
    //cout << str << endl;

    // 文件名
    char * sub2 = strrchr(filename, '/');
    sub2 += 1;
    //cout << sub2 << endl;
    StrReplace_(sub2, "'", "''");
   // cout << filename << endl;
    //sqlite3 *l_db=db;
    char *err_msg;
    int rc;
    //rc = sqlite3_open(dbname, &db);
    //if(rc)
    //{
    //    perror("open db failed\n");
    //    return false;
    //}
    char sql[1024] = {0};
    sprintf(sql, "insert into linux_All_file (id, file_name, uid, gid, size, st_atime, st_mtime, st_ctime, mac_address, event_task_id) values('%s', '%s', '%s', '%u', '%ld', '%s', '%s', '%s', '%s', '%s')",
        str, sub2, name, fileinfo.st_gid, fileinfo.st_size, fmt_atime, fmt_mtime, fmt_ctime, Mac, evenId);
    

    int ret = sqlite3_exec(db, sql, NULL, NULL, &err_msg);
    if(ret != 0)
    {
        perror(err_msg);
        sqlite3_free(err_msg);
        sqlite3_close(db);

        return false;
    }
    //sqlite3_close(db);

    return true;
}
